package cn.ryh.gat1400.utils;

/**
 * @Author: aiLin
 * @Date: 2020/8/24 14:24
 * @desc:
 */

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;


public class ImageWatermarkUtil {
    // 水印透明度
    private static float alpha = 0.2f;
    // 水印文字大小
    public static final int FONT_SIZE = 28;
    // 水印文字字体
    private static Font font = new Font("微软雅黑", Font.BOLD, FONT_SIZE);
    // 水印文字颜色
    private static Color color = Color.white;
    // 水印之间的间隔
    private static final int XMOVE = 80;
    // 水印之间的间隔
    private static final int YMOVE = 80;

    /**
     * 获取文本长度。汉字为1:1，英文和数字为2:1
     */
    private static int getTextLength(String text) {
        int length = text.length();
        for (int i = 0; i < text.length(); i++) {
            String s = String.valueOf(text.charAt(i));
            if (s.getBytes().length > 1) {
                length++;
            }
        }
        length = length % 2 == 0 ? length / 2 : length / 2 + 1;
        return length;
    }

    /**
     * 给图片添加水印文字
     *
     * @param logoText
     *            水印文字
     * @param srcImgPath
     *            源图片路径
     */
    public InputStream imageByText(String logoText, String srcImgPath) {
        return imageByText(logoText, srcImgPath, null);
    }

    /**
     * 给图片添加水印文字、可设置水印文字的旋转角度
     *
     * @param logoText
     * @param srcImgPath
     * @param degree
     */
    public InputStream imageByText(String logoText, String srcImgPath, Integer degree) {

        ByteArrayOutputStream os = new ByteArrayOutputStream();
        try {
            // 源图片
            Image srcImg = ImageIO.read(this.getImageStream(srcImgPath));
            int width = srcImg.getWidth(null);// 原图宽度
            int height = srcImg.getHeight(null);// 原图高度
            BufferedImage buffImg = new BufferedImage(srcImg.getWidth(null), srcImg.getHeight(null),
                    BufferedImage.TYPE_INT_RGB);
            // 得到画笔对象
            Graphics2D g = buffImg.createGraphics();
            // 设置对线段的锯齿状边缘处理
            g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
            g.drawImage(srcImg.getScaledInstance(srcImg.getWidth(null), srcImg.getHeight(null), Image.SCALE_SMOOTH), 0,
                    0, null);
            // 设置水印旋转
            if (null != degree) {
                g.rotate(Math.toRadians(degree), (double) buffImg.getWidth() / 2, (double) buffImg.getHeight() / 2);
            }
            // 设置水印文字颜色
            g.setColor(color);
            // 设置水印文字Font
            g.setFont(font);
            // 设置水印文字透明度
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
            g.drawString(logoText, width/2, height);

//			int x = -width / 2;
//			int y = -height / 2;
//			int markWidth = FONT_SIZE * getTextLength(logoText);// 字体长度
//			int markHeight = FONT_SIZE;// 字体高度
//
//			// 循环添加水印
//			while (x < width * 1.5) {
//				y = -height / 2;
//				while (y < height * 1.5) {
//					g.drawString(logoText, x, y);
//
//					y += markHeight + YMOVE;
//				}
//				x += markWidth + XMOVE;
//			}
            // 释放资源
            g.dispose();
            // 生成图片
            // 10、生成图片
            ImageIO.write(buffImg, "JPG", os);
            InputStream in = new ByteArrayInputStream(os.toByteArray());
            return in;
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != os)
                    os.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    /**
     * 获取网络图片流
     *
     * @param url
     * @return
     */
    public static InputStream getImageStream(String url) {
        try {
            HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection();
            connection.setReadTimeout(5000);
            connection.setConnectTimeout(5000);
            connection.setRequestMethod("GET");
            if (connection.getResponseCode() == HttpURLConnection.HTTP_OK) {
                InputStream inputStream = connection.getInputStream();
                return inputStream;
            }
        } catch (IOException e) {
            System.out.println("获取网络图片出现异常，图片路径为：" + url);
            e.printStackTrace();
        }
        return null;
    }

    public static void main(String[] args) {
        String url = "http://10.220.55.11/image/people_09e09716-a1f9-4e6b-b9a3-5f91d2be2c2d_people-photo4ad7d82f-7a43-4fe3-b5b0-361aaeda15fa.jpg?AWSAccessKeyId=BAJO7GSBGSQEXJ5OGIXV&Expires=1844674989&Signature=WQGjgX0wVbtYwJvl7iDRae6AdqE%3D\n";
        downloadPicture("F://MyDrivers//core//img//","555.jpg",url);
    }
    //链接url下载图片
    public static String downloadPicture(String savePath, String imageName, String urlList) {
        URL url = null;
        int imageNumber = 0;
        try {
            url = new URL(urlList);
            DataInputStream dataInputStream = new DataInputStream(url.openStream());
            savePath =  savePath+imageName;
            FileOutputStream fileOutputStream = new FileOutputStream(new File(savePath));
            ByteArrayOutputStream output = new ByteArrayOutputStream();

            byte[] buffer = new byte[1024];
            int length;
            while ((length = dataInputStream.read(buffer)) > 0) {
                output.write(buffer, 0, length);
            }
            byte[] context=output.toByteArray();
            fileOutputStream.write(output.toByteArray());
            dataInputStream.close();
            fileOutputStream.close();
        } catch (MalformedURLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return savePath;
    }

}
